/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package controller;

import java.io.IOException;
import java.io.PrintWriter;
import java.sql.SQLException;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import model.DBFunction;

/**
 *
 * @author ThanhTrung
 */
public class Servlet extends HttpServlet {

    /** 
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        response.setContentType("text/html;charset=UTF-8");
        PrintWriter out = response.getWriter();
        DBFunction dbFunction = new DBFunction();
        try {
            if (request.getParameter("action").toString().equalsIgnoreCase("login")) {
                doLogin(request, response);
            } else if (request.getParameter("action").toString().equalsIgnoreCase("register")) {
                doRegister(request, response);
            } else  if(request.getParameter("action").toString().equalsIgnoreCase("product"))
            {
                doList(request,response);
            }else {
                response.sendRedirect("login.jsp");
            }
        }catch(SQLException ex){
            ex.printStackTrace();
        }catch(ClassNotFoundException ex){
            ex.printStackTrace();
        }catch(Exception ex){
            ex.printStackTrace();
        }
        finally {
            out.close();
        }
    }
    public void doList(HttpServletRequest request,HttpServletResponse response) throws IOException{
        PrintWriter out = response.getWriter();
        out.print("<h1>aloalaoalaoalaolaoalaolaoalaoalaolaoa");
        response.setHeader("status", "200");
    }
    private void doRegister(HttpServletRequest request, HttpServletResponse response) throws Exception{
        PrintWriter out = response.getWriter();
        DBFunction dbFunction = new DBFunction();
        boolean flag = false;
        String username = request.getParameter("txtUsername");
        String password = request.getParameter("txtPassword");
        String repassword = request.getParameter("txtRepassword");


        if (username.equals("") || password.equals("")) {
            RequestDispatcher dispatcher = request.getRequestDispatcher("register.jsp");
            request.setAttribute("message", "<font color=\"red\">Username and password can not blank.</font>");
            dispatcher.forward(request, response);
        } else if (!repassword.equals(password)) {
            RequestDispatcher dispatcher = request.getRequestDispatcher("register.jsp");
            request.setAttribute("message", "<font color=\"red\">Password and repassword do not match.</font>");
            dispatcher.forward(request, response);
        } else {
            flag = dbFunction.register(username, password);
            if (flag == true) {

                out.println("<html>");
                out.println("<head>");
                out.println("<title>WPSJ_Lab2</title>");
                out.println("</head>");
                out.println("<body>");
                out.println("<center>");
                out.println("<h1>Register successfully.</h1>");
                out.println("</center>");
                out.println("</body>");
                out.println("</html>");
            } else {
                RequestDispatcher dispatcher = request.getRequestDispatcher("register.jsp");
                request.setAttribute("message", "<font color=\"red\">Register fail.</font>");
            }
        }

    }

    private void doLogin(HttpServletRequest request, HttpServletResponse response) throws SQLException,IOException, ClassNotFoundException,ServletException {
        response.setContentType("text/html;charset=UTF-8");
        PrintWriter out = response.getWriter();
        DBFunction dbFunction = new DBFunction();
        String username = request.getParameter("txtUsername");
        String password = request.getParameter("txtPassword");

        boolean flag = dbFunction.checkLogin(username, password);

        if (flag == true) {
            out.println("<html>");
            out.println("<head>");
            out.println("<title>WPSJ_Lab2</title>");
            out.println("</head>");
            out.println("<body>");
            out.println("<center>");
            out.println("<h1>Login successfully. </h1>");
            out.println("</center>");
            out.println("</body>");
            out.println("</html>");
        } else {
            RequestDispatcher dispatcher = request.getRequestDispatcher("login.jsp");
            request.setAttribute("message", "<font color=\"red\">Login fail. Username or password is invalid.</font>");
            dispatcher.forward(request, response);

        }
    }

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /** 
     * Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
   
            processRequest(request, response);

    }

    /** 
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /** 
     * Returns a short description of the servlet.
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>
}
